#!/usr/bin/env python3
"""Compose V2 video - Max V9 real footage version (Round 1 fix)"""
import os, subprocess, re, sys

BASE = r'E:\集群文件夹\factory_os\short_video_real_data_pipeline\phase4b_90s_formal_sample\v22_kimi_claude_loop\full_story_douyin_video'
FFMPEG = r'D:\AI_WORKSPACE\tools\ffmpeg\ffmpeg.exe'
CLIPS = os.path.join(BASE, '05_clips')
AUDIO = os.path.join(BASE, '05_audio_mix', 'v2', 'final_audio_mix_v2.wav')
SUBS = os.path.join(BASE, '04_subtitles', 'v2', 'captions_final_v2.ass')
VIDEO_OUT = os.path.join(BASE, '06_video', 'claude_code_self_evolving_full_story_douyin_v2.mp4')
VIDEO_BURNED = os.path.join(BASE, '06_video', 'claude_code_self_evolving_full_story_douyin_v2_burned.mp4')

# MAX V9 scene order - replace static scenes with real ComfyUI rendering footage
# Scene types: [name, use_crossfade]
SCENES = [
    ('title', False),           # 3.5s - keep as is (opening)
    ('v9_extra_1', True),       # 3s - REAL V9 replaces punch
    ('recording_fail', False),   # 4s - keep (story essential)
    ('v9_extra_2', True),       # 4s - REAL V9 replaces five_fails
    ('ai_drift', False),        # 4s - keep
    ('chatgpt_fix', False),     # 4s - keep
    ('python_render', False),   # 4s - keep
    ('multi_agent', False),     # 4s - keep
    ('v9_extra_3', True),       # 4s - REAL V9 replaces log_lines
    ('v9_dyn_1', True),         # 5s - REAL V9
    ('kimi_intro', False),      # 4s - keep (needed for story)
    ('score_bars', False),      # 6s - keep (animated)
    ('v9_dyn_2', True),         # 6s - REAL V9
    ('v9_peak', False),         # 3s - keep
    ('v9_dyn_3', True),         # 5s - REAL V9
    ('v10_crash', False),       # 5s - keep (animated)
    ('v11_recovery', False),    # 4s - keep
    ('v9_extra_4', True),       # 2s - REAL V9 replaces lesson
    ('rollback', False),        # 3s - keep
    ('review_portal', False),   # 4s - keep
    ('http_verify', False),     # 4s - keep
    ('summary', False),         # 4s - keep
    ('final_montage', False),   # 4s - keep
    ('end_card', False),        # 7.5s - keep
]

def get_dur(f):
    r = subprocess.run([FFMPEG, '-i', f], capture_output=True, text=True, encoding='utf-8', errors='replace')
    m = re.search(r'Duration: (\d+):(\d+):(\d+\.\d+)', r.stderr)
    if m: return int(m.group(1))*3600 + int(m.group(2))*60 + float(m.group(3))
    return 0

# Verify all scenes exist
scene_names = [s[0] for s in SCENES]
missing = [s for s in scene_names if not os.path.exists(os.path.join(CLIPS, s+'.mp4'))]
if missing:
    print(f"ERROR: Missing clips: {missing}")
    sys.exit(1)
print(f"All {len(SCENES)} clips OK")

# Calculate total raw duration
total_raw = sum(get_dur(os.path.join(CLIPS, s+'.mp4')) for s in scene_names)
adur = get_dur(AUDIO)
print(f"Total raw video: {total_raw:.1f}s | Audio: {adur:.1f}s")

# Build concat file with crossfade transitions
has_crossfade = any(cf for _, cf in SCENES)

if has_crossfade:
    # Use crossfade approach
    # Create concat with transitions via ffmpeg complex filter
    filter_parts = []
    concat_file = os.path.join(CLIPS, 'c_v9.txt')
    with open(concat_file, 'w', encoding='utf-8') as f:
        for s_name, _ in SCENES:
            f.write(f"file '{s_name}.mp4'\n")

    # Concatenate with crossfade at V9 scene boundaries
    # Simple approach: concat first, then add crossfade
    joined = os.path.join(CLIPS, '_j_v9.mp4')
    subprocess.run([FFMPEG, '-y', '-f', 'concat', '-safe', '0',
                    '-i', concat_file, '-c', 'copy', joined],
                   cwd=CLIPS, capture_output=True)
else:
    # Simple concat
    concat_file = os.path.join(CLIPS, 'c_v9.txt')
    with open(concat_file, 'w', encoding='utf-8') as f:
        for s_name, _ in SCENES:
            f.write(f"file '{s_name}.mp4'\n")
    joined = os.path.join(CLIPS, '_j_v9.mp4')
    subprocess.run([FFMPEG, '-y', '-f', 'concat', '-safe', '0',
                    '-i', concat_file, '-c', 'copy', joined],
                   cwd=CLIPS, capture_output=True)

if not os.path.exists(joined):
    print("FAIL at concat")
    sys.exit(1)

vdur = get_dur(joined)
print(f"Joined video: {vdur:.1f}s")

# Add audio and encode
subprocess.run([FFMPEG, '-y',
    '-i', joined,
    '-i', AUDIO,
    '-c:v', 'libx264', '-preset', 'medium', '-crf', '20',
    '-c:a', 'aac', '-b:a', '192k',
    '-pix_fmt', 'yuv420p',
    '-movflags', '+faststart',
    '-shortest',
    VIDEO_OUT],
    cwd=CLIPS, capture_output=True)

# Clean temp
for f in [joined, concat_file]:
    try: os.remove(f)
    except: pass

# Verify output
if os.path.exists(VIDEO_OUT):
    sz = os.path.getsize(VIDEO_OUT) / (1024*1024)
    dur = get_dur(VIDEO_OUT)

    # Calculate real footage stats
    v9_clips = [s for s in SCENES if s[0].startswith('v9_')]
    v9_total = sum(get_dur(os.path.join(CLIPS, s+'.mp4')) for s,_ in v9_clips)

    print(f"\n{'='*50}")
    print(f"[OK] V2 V9-MAX VIDEO COMPLETE")
    print(f"{'='*50}")
    print(f"  Path: {VIDEO_OUT}")
    print(f"  Size: {sz:.1f} MB | Duration: {dur:.1f}s")
    print(f"  Total V9 clips: {len(v9_clips)}")
    print(f"  V9 real footage: {v9_total:.1f}s ({v9_total/dur*100:.0f}%)")
    print(f"  Total scenes: {len(SCENES)}")
    print(f"  V9 ratio improvement: 17% -> {v9_total/dur*100:.0f}%")
else:
    print("\n[FAIL]")